import jieba # 使用jieba进行分词
from rouge import Rouge
def jieba_tokenize(text):
    return ' '.join(jieba.cut(text))



def rouge_score_evaluation(generated_summary, reference_summary):
    # 初始化 Rouge 评估器
    rouge = Rouge()
    # 分词后的生成摘要和参考摘要
    hyps = jieba_tokenize(generated_summary)
    refs = jieba_tokenize(reference_summary)
    # 计算单个摘要的 ROUGE 分数
    scores = rouge.get_scores(hyps=hyps, refs=refs)
    # 打印 ROUGE 分数
    return scores

